package com.xkcyy.one.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.xkcyy.one.entity.AgentCollaboration;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Select;

import java.util.List;

/**
 * 智能体协作关联数据访问层
 * 
 * @author yuand
 */
@Mapper
public interface AgentCollaborationMapper extends BaseMapper<AgentCollaboration> {

    /**
     * 根据主智能体ID查询关联的从智能体ID列表
     *
     * @param agentId 主智能体ID
     * @return 从智能体ID列表
     */
    @Select("SELECT sub_agent_id FROM agent_collaboration WHERE agent_id = #{agentId}")
    List<Long> selectSubAgentIdsByAgentId(Long agentId);

    /**
     * 根据从智能体ID查询关联的主智能体ID列表
     *
     * @param subAgentId 从智能体ID
     * @return 主智能体ID列表
     */
    @Select("SELECT agent_id FROM agent_collaboration WHERE sub_agent_id = #{subAgentId}")
    List<Long> selectAgentIdsBySubAgentId(Long subAgentId);
}